package com.loveshare.imgshow.util;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;

public class DownloadFileUtil4File {

	public static ArrayList<File> files4Path;
	/** 
     * 从网络Url中下载文件 
     * @param urlStr 
     * @param fileName 
     * @param savePath 
     * @throws IOException 
     */  
    public static void  downLoadFromUrl(String urlStr,String fileName,String savePath) throws IOException{  
        try {
        	URL url = new URL(urlStr);    
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();    
                    //设置超时间为3秒  
            conn.setConnectTimeout(3*1000);  
            //防止屏蔽程序抓取而返回403错误  
            conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");  
      
            //得到输入流  
            InputStream inputStream = conn.getInputStream();    
            //获取自己数组  
            byte[] getData = readInputStream(inputStream);      
            //文件保存位置  
            File saveDir = new File(savePath);  
            if(!saveDir.exists()){  
                saveDir.mkdir();  
            }  
            File file = new File(saveDir+File.separator+fileName);      
            FileOutputStream fos = new FileOutputStream(file);       
            fos.write(getData);   
            if(fos!=null){  
                fos.close();    
            }  
            if(inputStream!=null){  
                inputStream.close();  
            } 
            System.out.println("info:"+url+" download success");
		} catch (Exception e) {
			System.out.println(e);
		} 
    }  
  
    /** 
     * 从输入流中获取字节数组 
     * @param inputStream 
     * @return 
     * @throws IOException 
     */  
    public static  byte[] readInputStream(InputStream inputStream) throws IOException {    
        byte[] buffer = new byte[1024];    
        int len = 0;    
        ByteArrayOutputStream bos = new ByteArrayOutputStream();    
        while((len = inputStream.read(buffer)) != -1) {    
            bos.write(buffer, 0, len);    
        }    
        bos.close();    
        return bos.toByteArray();    
    }
    
    /**
     * 获取文件的数据流
     * 返回null代表无效文件
     * @param urlStr
     * @return
     */
    public static InputStream getFileInputStream(String urlStr){
    	try {
    		URL url = new URL(urlStr);    
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();    
            // 设置超时间为3秒  
            conn.setConnectTimeout(3*1000);  
            // 防止屏蔽程序抓取而返回403错误  
            conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");  
            // 获取文件大小
            if (conn.getContentLength() <= 0) return null;
            // 得到输入流 
            InputStream inputStream = conn.getInputStream();
            return inputStream;
		} catch (Exception e) {
		}
    	return null;
        
    }
  
    
    /**
     * 获取文件夹下所有文件
     * @param fileList
     * @param path
     */
    public static void getFiles(ArrayList<File> fileList, String path) {
        File[] allFiles = new File(path).listFiles();
        for (int i = 0; i < allFiles.length; i++) {
            File file = allFiles[i];

            if (file.isFile()) {
                    fileList.add(file);
                    files4Path.add(file);
            } else  {
                getFiles(fileList, file.getAbsolutePath());
            }
        }
    }
    
    /**
     * 获取整个文件夹下所有文件包括文件里面
     * @return
     */
    public static ArrayList<File> getFile4Path(String path){
    	 files4Path= new ArrayList<File>();
         getFiles(files4Path, path);
         return files4Path;
    }
}
